package com.demo.mqtt.common.client;

import com.demo.mqtt.common.constants.TopictConstants;
import com.demo.mqtt.common.utils.HexConvertUtil;
import com.demo.mqtt.picker.entity.PickerInfo;
import lombok.extern.slf4j.Slf4j;
import org.eclipse.paho.client.mqttv3.*;
import org.eclipse.paho.client.mqttv3.persist.MemoryPersistence;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;
import com.demo.mqtt.common.properties.MqttProperties;

import java.util.UUID;
import java.util.concurrent.atomic.AtomicInteger;


/**
 * @program: mqtt
 * @description: mqtt客户端
 * @author: liuYunKai
 * @create: 2021-03-23 13:06
 */
@Slf4j
@Component
public class MqttConnectionUtils implements TopictConstants {

    @Autowired
    private MqttCallback mqttCallback;
    @Autowired
    private MqttProperties mqttProperties;


    public static final String TOPIC1 = "/publish/heart_beat";

//    public static final String TOPIC1 = "/publish/features_datas";

//    public static final String TOPIC1 = "/publish/#";

//    public static final String TOPIC1 = "mqtt";

    private static String clientid;
    private MqttClient client;
    private MqttConnectOptions options;

    // 开始连接
    public void start() {
        try {
            clientid = UUID.randomUUID().toString().trim().replaceAll("-", "");

            // host为主机名，clientid即连接MQTT的客户端ID，一般以唯一标识符表示，MemoryPersistence设置clientid的保存形式，默认为以内存保存
            client = new MqttClient(mqttProperties.getBrokerUrl(), clientid, new MemoryPersistence());
            // MQTT的连接设置
            options = new MqttConnectOptions();
            // 设置是否清空session,这里如果设置为false表示服务器会保留客户端的连接记录，设置为true表示每次连接到服务器都以新的身份连接
            options.setCleanSession(false);
            // 设置连接的用户名
            options.setUserName(mqttProperties.getUsername());
            // 设置连接的密码
            options.setPassword(mqttProperties.getPassword().toCharArray());


            // 设置超时时间 单位为秒
            options.setConnectionTimeout(10);
            // 设置会话心跳时间 单位为秒 服务器会每隔1.5*20秒的时间向客户端发送个消息判断客户端是否在线，但这个方法并没有重连的机制
            options.setKeepAliveInterval(20);
            // 设置断开后重新连接
            options.setAutomaticReconnect(true);
            // 设置回调
            client.setCallback(mqttCallback);
//            MqttTopic topic = client.getTopic(TOPIC1);
            // setWill方法，如果项目中需要知道客户端是否掉线可以调用该方法。设置最终端口的通知消息 遗嘱
//            options.setWill(topic, "close".getBytes(), QosEnum.QOS2.value(), true);
            client.connect(options);
            // 订阅消息
            sub(SUB_TOPIC);
        } catch (Exception e) {
            e.printStackTrace();
        }
    }

    /**
     * 关闭MQTT连接
     */
    public void close() throws MqttException {
        client.close();
        client.disconnect();
    }

    /**
     * 向某个主题发布消息 默认qos：1
     *
     * @param topic:发布的主题
     * @param msg：发布的消息
     */
    public void pub(String topic, String msg) throws MqttException {
        MqttMessage mqttMessage = new MqttMessage();
        mqttMessage.setPayload(msg.getBytes());
//        MqttTopic mqttTopic = client.getTopic(topic);
//        MqttDeliveryToken token = mqttTopic.publish(mqttMessage);
//        token.waitForCompletion();


        if (client.isConnected()) {

            client.publish(topic, mqttMessage);

        }

    }

    /**
     * 向某个主题发布消息
     *
     * @param topic: 发布的主题
     * @param msg:   发布的消息
     * @param qos:   消息质量    Qos：0、1、2
     */
    public void pub(String topic, String msg, int qos) throws MqttException {
        MqttMessage mqttMessage = new MqttMessage();
        mqttMessage.setQos(qos);
        mqttMessage.setPayload(msg.getBytes());
//        MqttTopic mqttTopic = client.getTopic(topic);
//        MqttDeliveryToken token = mqttTopic.publish(mqttMessage);
//        token.waitForCompletion();


        if (client.isConnected()) {

            client.publish(topic, mqttMessage);

        }
    }

    /**
     * 向某个主题发布消息
     *
     * @param topic: 发布的主题
     * @param msg:   发布的消息 byte[]
     * @param qos:   消息质量    Qos：0、1、2
     */
    public void pub4Byte(String topic, byte[] msg, int qos) throws MqttException {
        MqttMessage mqttMessage = new MqttMessage();
        mqttMessage.setQos(qos);
        mqttMessage.setPayload(msg);

        if (client.isConnected()) {

            client.publish(topic, mqttMessage);

        }
    }

    /**
     * 向 通用应答主题 发布消息
     *
     * @param topic: 发布的主题
     * @param msg:   发布的消息
     * @param qos:   消息质量    Qos：0、1、2
     */
    public void pubToNormalAck(String msg, int qos) throws MqttException {
        MqttMessage mqttMessage = new MqttMessage();
        mqttMessage.setQos(qos);
        mqttMessage.setPayload(HexConvertUtil.hexStrToByteArrs(msg));

        if (client.isConnected()) {

            client.publish(SUB_NARMAL_ACK, mqttMessage);

        }
    }


    /**
     * 订阅某一个主题 ，此方法默认的的Qos等级为：1
     *
     * @param topic 主题
     */
    public void sub(String topic) throws MqttException {
        client.subscribe(topic);
    }

    /**
     * 订阅某一个主题，可携带Qos
     *
     * @param topic 所要订阅的主题
     * @param qos   消息质量：0、1、2
     */
    public void sub(String topic, int qos) throws MqttException {
        client.subscribe(topic, qos);
    }


    /**
     * 下发传感器配置
     *
     * @return
     */
    public static AtomicInteger count = new AtomicInteger(1);

    public Boolean sendSensorSetting(PickerInfo pickerInfo) {

        StringBuilder stringBuilder = new StringBuilder();

        // 协议头 header + 协议版本号 version
        stringBuilder.append("aaaa01");
        // 设备地址 ip
        stringBuilder.append("000000000000");
        // 序列号 sn
        String hesStr = String.format("%04x", count.intValue());
        stringBuilder.append(hesStr.substring(2, 4) + hesStr.substring(0, 2));
        // 通讯命令 order
        stringBuilder.append("22");
        // 数据长度 length
        stringBuilder.append("02000000");

        /**-------------数据内容 begin-------------*/
        //参数下发时间
        stringBuilder.append("14140909081e00");
        //参数类型数量
        stringBuilder.append("0f");
        //高频普通波形-宽频上传时间
        stringBuilder.append("0101");
        //高频普通波形-宽频上传时间 value
        stringBuilder.append("00000005");
        //低频普通波形-宽频上传时间
        stringBuilder.append("0102");
        //低频普通波形-宽频上传时间 value
        stringBuilder.append("00000005");
        //低频普通波形-低频上传时间
        stringBuilder.append("0103");
        //低频普通波形-低频上传时间 value
        stringBuilder.append("00000005");
        //长波形- 宽频上传时间
        stringBuilder.append("0104");
        //长波形- 宽频上传时间 value
        stringBuilder.append("00000005");

        //高频普通波形-宽频上传数据长度
        stringBuilder.append("0201");
        //高频普通波形-宽频上传数据长度 value
        stringBuilder.append("00000004");
        //低频普通波形-宽频上传数据长度
        stringBuilder.append("0202");
        //低频普通波形-宽频上传数据长度 value
        stringBuilder.append("00000004");
        //低频普通波形-低频上传数据长度
        stringBuilder.append("0203");
        //低频普通波形-低频上传数据长度 value
        stringBuilder.append("00000004");
        //长波形- 宽频上传数据长度
        stringBuilder.append("0204");
        //长波形- 宽频上传数据长度 value
        stringBuilder.append("00000004");

        //特征值采集周期
        stringBuilder.append("0301");
        //特征值采集周期 value
        stringBuilder.append("00000006");
        //特征值上传周期
        stringBuilder.append("0302");
        //特征值上传周期 value
        stringBuilder.append("00000006");
        //普通波形采集周期
        stringBuilder.append("0304");
        //普通波形采集周期 value
        stringBuilder.append("00000006");
        //长波形采集周期
        stringBuilder.append("0305");
        //长波形采集周期 value
        stringBuilder.append("00000006");

        //高加总值报警三级门限
        stringBuilder.append("0401");
        //高加总值报警三级门限 value
        stringBuilder.append("03000006");
        //高加总值报警四级门限
        stringBuilder.append("0402");
        //高加总值报警四级门限 value
        stringBuilder.append("05000006");
        //低加总值报警三级门限
        stringBuilder.append("0403");
        //低加总值报警三级门限 value
        stringBuilder.append("02000106");

        /**-------------数据内容 end-------------*/
        // 校验和 sum
        stringBuilder.append("58");
        // 协议尾 end
        stringBuilder.append("bb");


        try {
            pubToSetObjParam(stringBuilder.toString(), 1);
        } catch (MqttException e) {
            log.error("设置传感器参数报错：{}", e.getMessage());
        }

        count.getAndIncrement();
        return true;
    }

    /**
     * 设置传感器参数
     *
     * @param toString
     * @param i
     */
    private void pubToSetObjParam(String msg, int qos) throws MqttException {
        MqttMessage mqttMessage = new MqttMessage();
        mqttMessage.setQos(qos);
        mqttMessage.setPayload(HexConvertUtil.hexStrToByteArrs(msg));

        if (client.isConnected()) {

            client.publish(SUB_SET_OBJ_PARAM, mqttMessage);

        }

    }
}
